Sum of digits of number using sum() function in Python